package com.spring.boot.demo.config;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.spring.boot.demo.anno.LogRecord;
import com.spring.boot.demo.pojo.LogRecordEntity;
import com.spring.boot.demo.util.HttpRequestUtil;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.*;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.lang.reflect.Method;
import java.util.UUID;

/**
 * @author ZhuYouBin
 * @version 1.0.0
 * @Date: 2022/11/2 12:52
 * @Description 自定义日志切面
 */
// 标记当前类是一个切面类
@Aspect
// 将当前类放入IOC容器
@Component
public class LogAspect {

    /**
     * 创建线程局部变量
     */
    private ThreadLocal<LogRecordEntity> threadLocal = new ThreadLocal<>();

    /**
     * 定义切入点，这里我们使用AOP切入自定义【@LogRecord】注解的方法
     */
    @Pointcut("@annotation(com.spring.boot.demo.anno.LogRecord)")
    public void pointCut() {}

    /**
     * 前置通知,【执行Controller方法之前】执行该通知方法
     */
    @Before("pointCut()")
    public void beforeAdvice() {
        System.out.println("前置通知......"); // TODO delete
    }

    /**
     * 后置通知,【Controller方法执行完成,返回方法的返回值之前】执行该通知方法
     */
    @After("pointCut()")
    public void afterAdvice() {
        System.out.println("后置通知......"); // TODO delete
    }

    /**
     * 环绕通知,执行Controller方法的前后执行
     * @param joinPoint 连接点
     */
    @Around("pointCut()")
    public Object aroundAdvice(ProceedingJoinPoint joinPoint) throws Throwable {
        // 请求开始时间
        String requestTime = String.valueOf(System.currentTimeMillis());
        System.out.println("环绕通知之前....."); // TODO delete
        // 获取当前请求对象
        HttpServletRequest request = HttpRequestUtil.getRequest();
        if (request == null) {
            return null;
        }
        // 获取请求相关信息
        LogRecordEntity entity = new LogRecordEntity();
        entity.setId(UUID.randomUUID().toString().replace("-", ""));
        entity.setPath(request.getRequestURI());
        entity.setMethod(request.getMethod());
        entity.setRequestIp(request.getRemoteHost());
        entity.setRequestTime(requestTime);

        // 反射获取调用方法
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        Method method = signature.getMethod();
        if (method.isAnnotationPresent(LogRecord.class)) {
            // 获取注解信息
            LogRecord annotation = method.getAnnotation(LogRecord.class);
            entity.setOpName(annotation.opName());
        }
        // 获取全限定类名称
        String name = method.getName();

        // 获取请求参数
        String inputParam = JSONObject.toJSONString(joinPoint.getArgs());
        entity.setInputParam(inputParam);
        // 设置局部变量
        threadLocal.set(entity);

        // 调用Controller方法
        Object ret = joinPoint.proceed();
        System.out.println("环绕通知之后....."); // TODO delete
        return ret;
    }

    /**
     * 返回值通知,Controller执行完成之后,返回方法的返回值时候执行
     * @param ret 返回值的名称
     */
    @AfterReturning(pointcut = "pointCut()", returning = "ret")
    public Object afterReturning(Object ret) {
        System.out.println("返回值通知......ret=" + ret); // TODO delete
        // 获取日志实体对象
        LogRecordEntity entity = this.getEntity();
        String outputParam = JSON.toJSONString(ret);
        entity.setOutputParam(outputParam); // 保存响应参数
        entity.setStatus("成功"); // 设置成功标识
        // TODO 这里就可以做一些持久胡操作，例如：保存日志到数据表里面
        // 一定要删除 ThreadLocal 变量
        threadLocal.remove();
        System.out.println(entity); // TODO delete
        return ret;
    }

    /**
     * 异常通知,当Controller方法执行过程中出现异常时候,执行该通知
     * @param ex 异常名称
     */
    @AfterThrowing(pointcut = "pointCut()", throwing = "ex")
    public void throwingAdvice(Throwable ex) {
        System.out.println("异常通知......"); // TODO delete
        // 获取日志实体对象
        LogRecordEntity entity = this.getEntity();
        StringWriter errorMsg = new StringWriter();
        ex.printStackTrace(new PrintWriter(errorMsg, true));
        entity.setErrorMsg(errorMsg.toString()); // 保存响应参数
        entity.setStatus("失败"); // 设置成功标识
        // TODO 这里就可以做一些持久胡操作，例如：保存日志到数据表里面
        // 一定要删除 ThreadLocal 变量
        threadLocal.remove();
        System.out.println(entity); // TODO delete
    }

    /****************************************************/

    private LogRecordEntity getEntity() {
        // 获取局部变量
        LogRecordEntity entity = threadLocal.get();
        long start = Long.parseLong(entity.getRequestTime());
        long end = System.currentTimeMillis();
        // 获取响应时间、耗时
        entity.setCostTime((end - start) + "ms");
        entity.setResponseTime(String.valueOf(end));
        return entity;
    }
}
